/*
 * To change this template, choose Tools | Templates
 * and open the template in the editor.
 */
package edu.Ufes.exeSwing.Collection;

import edu.Ufes.exeSwing.Persistence.ContaPersistence;
import edu.Ufes.exeSwing.Model.Conta;
import java.util.Iterator;
import java.util.TreeSet;

/**
 *
 * @author bhetynnacatalunha
 */
public class ContaCollection {

    ContaPersistence contaPersistencia;

    public ContaCollection(ContaPersistence pContaPersistencia) {
        this.contaPersistencia = pContaPersistencia;
    }

    public void incluirConta(Conta pConta) {
        TreeSet<Conta> contas = contaPersistencia.recuperarContas();
        contas.add(pConta);
        contaPersistencia.salvarConta(contas);
    }

    public void excluirConta(Conta pConta) {
        TreeSet<Conta> contas = contaPersistencia.recuperarContas();
        contas.remove(pConta);
        contaPersistencia.salvarConta(contas);
    }

    public Conta getUmaConta(String pCodigo) {
        Conta retorno = null;
        Conta conta;
        TreeSet<Conta> contas = contaPersistencia.recuperarContas();
        Iterator<Conta> itContas = contas.iterator();
        while (itContas.hasNext()) {
            conta = itContas.next();
            if (conta.getCodigo().equals(pCodigo)) {
                retorno = conta;
            }
        }
        return retorno;
    }

    public TreeSet<Conta> getTodasContas() {
        TreeSet<Conta> contas = contaPersistencia.recuperarContas();
        return contas;
    }
    
    public void editarConta(Conta pConta, String pCodigo){
        TreeSet<Conta> contas = contaPersistencia.recuperarContas();
        Conta a = this.getUmaConta(pCodigo);
        this.excluirConta(a);
        contas.add(pConta);
        contaPersistencia.salvarConta(contas);
    
    }
}
